/**
 * Definition for singly-linked list.
 * function ListNode(val, next) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.next = (next===undefined ? null : next)
 * }
 */
/**
 * @param {ListNode} head
 * @param {number} x
 * @return {ListNode}
 */
var partition = function (head, x) {
  // 创建两个虚拟头节点，分别用于存储小于x和大于等于x的节点
  const dummy1 = new ListNode(0); // 小于x的链表
  const dummy2 = new ListNode(0); // 大于等于x的链表

  let p1 = dummy1; // 指向小于x链表的当前节点
  let p2 = dummy2; // 指向大于等于x链表的当前节点

  let current = head; // 遍历原链表的指针

  // 遍历原链表，将节点分配到两个链表中
  while (current !== null) {
    if (current.val < x) {
      p1.next = current;
      p1 = p1.next;
    } else {
      p2.next = current;
      p2 = p2.next;
    }
    current = current.next;
  }

  // 连接两个链表
  p1.next = dummy2.next; // 将小于x的链表尾部连接到大于等于x的链表头部
  p2.next = null; // 大于等于x的链表尾部置为null，防止形成环

  return dummy1.next; // 返回新链表的头节点
};